# __init__.py

# 1. Importer les classes avec leurs nouveaux noms depuis votre fichier principal
from .ensemble_upscaler import EnsembleSuperRes, SaveImageAdvanced

# 2. Déclarer les "mappings" que ComfyUI utilisera pour trouver et nommer les noeuds
NODE_CLASS_MAPPINGS = {
    "EnsembleSuperRes_Orion4D": EnsembleSuperRes,
    "SaveImageAdvanced_Orion4D": SaveImageAdvanced, # <-- On utilise le nouveau nom ici
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "EnsembleSuperRes_Orion4D": "Ensemble Super-Resolution (HDR)",
    "SaveImageAdvanced_Orion4D": "Save Image Advanced (16b/32f)", # <-- Et le nouveau nom d'affichage ici
}

# 3. Exposer ces mappings à ComfyUI
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']

print("[Orion4D] Nodes chargés : EnsembleSuperRes (HDR) et SaveImageAdvanced.")